977. Squares of a Sorted Array - LeetCode Solution

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        p1 = 0
        p2 = len(nums) - 1
        output = []
        
        for i in range(len(nums)):
            if(nums[p1]**2>=nums[p2]**2):
                output.append(nums[p1]**2)
                p1+=1
            else:
                output.append(nums[p2]**2)
                p2-=1
        
        return output[::-1]